# sk-476b894aa2d74474b77d4dfa91633f72
import os
from langchain.chat_models import init_chat_model
from dotenv import load_dotenv

# 加载.env文件
load_dotenv()

model = init_chat_model(
    model='deepseek-reasoner', # deepseek-chat表示调用DeepSeek-v3模型，deepseek-reasoner表示调用DeepSeek-R1模型，
    model_provider='deepseek',# 模型提供商写deepseek
    # api_key="sk-476b894aa2d74474b77d4dfa91633f72", #你注册的deepseek api_key
    api_key=os.environ.get("DEEPSEEK_API_KEY"),
)

question="地球是圆的还是方的"

result = model.invoke(question)
print(result)
